package com.xiao.test;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;

import java.util.Collections;

/**
 * 需要导入mybatis-plus依赖
 * 和代码生成器依赖
 *  <!-- mybatis-plus -->
 *         <dependency>
 *             <groupId>com.baomidou</groupId>
 *             <artifactId>mybatis-plus-boot-starter</artifactId>
 *             <version>3.5.1</version>
 *         </dependency>
 *  <!-- 代码生成器 -->
 *         <dependency>
 *             <groupId>com.baomidou</groupId>
 *             <artifactId>mybatis-plus-generator</artifactId>
 *             <version>3.5.1</version>
 *         </dependency>
 *         <dependency>
 *             <groupId>org.apache.velocity</groupId>
 *             <artifactId>velocity</artifactId>
 *             <version>1.7</version>
 *         </dependency>
 */

public class CodeGenerator {

    public static void main(String[] args) {
        generate();
    }

    private static void generate() {
        FastAutoGenerator.create("你的数据库url", "root", "root")
                .globalConfig(builder -> {
                    builder.author("xiaobo") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("输出目录"); // 指定输出目录   D:\xxx\xxx\xxx\src\main\
                })
                .packageConfig(builder -> {
                    builder.parent("您的父包名") // 设置父包名   com.xxx
                            .moduleName(null) // 设置父包模块名                         D:\xxx\xxx\xxx\src\main\resources\mapper\\
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "mapper要输出的目录")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.entityBuilder().enableLombok();
//                    builder.mapperBuilder().enableMapperAnnotation().build();
                    builder.controllerBuilder().enableHyphenStyle()  // 开启驼峰转连字符
                            .enableRestStyle();  // 开启生成@RestController 控制器
                    builder.addInclude("你数据库的表明  多个请用,隔开","表一","表二") // 设置需要生成的表名
                            .addTablePrefix("tb_", "sys_"); // 设置过滤表前缀
                })
//                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();

    }
}
